

package com.hazelcast.spi.discovery.impl;

import com.hazelcast.cluster.Address;
import com.hazelcast.spi.discovery.DiscoveryNode;
import com.hazelcast.spi.discovery.DiscoveryStrategy;
import com.hazelcast.spi.discovery.integration.DiscoveryService;

import java.util.Map;
import java.util.Set;

import static com.hazelcast.internal.util.Preconditions.checkNotNull;

/**
 * Discovery service with a predefined and supplied discovery strategy. All methods delegate to the provided strategy.
 */
public class PredefinedDiscoveryService implements DiscoveryService {
    private final DiscoveryStrategy strategy;

    public PredefinedDiscoveryService(DiscoveryStrategy strategy) {
        this.strategy = checkNotNull(strategy, "Discovery strategy should be not-null");
    }

    @Override
    public void start() {
        strategy.start();
    }

    @Override
    public Iterable<DiscoveryNode> discoverNodes() {
        return strategy.discoverNodes();
    }

    @Override
    public Map<String, String> discoverLocalMetadata() {
        return strategy.discoverLocalMetadata();
    }

    @Override
    public void markEndpointAsUnhealthy(Address address) {
        strategy.markEndpointAsUnhealthy(address);
    }

    @Override
    public Set<Address> getUnhealthyEndpoints() {
        return strategy.getUnhealthyEndpoints();
    }

    @Override
    public void destroy() {
        strategy.destroy();
    }

    public DiscoveryStrategy getStrategy() {
        return strategy;
    }
}
